#include "worker.h"
#include <mysql/mysql.h>

class MysqlManagement {
public:
  // 构造函数
  MysqlManagement();
  // 连接数据库
  int connectDB();

  // 是否在数据库中
  bool isInDB(int id);

  // 添加数据库
  // 返回值为1 添加成功
  // 返回值为0 添加失败
  int addDB(Worker *newWorker);

  // 显示数据库
  int showDB(MYSQL_RES *result);

  // 删除员工数据库
  int deleteDB(int number);

  // 修改员工
  int alterDB(int id, Worker *updatedWorker);

  // 排序数据库
  int sortDB();

  // 清空数据库
  int clearDB();

public:
  MYSQL *db;
};
